# 生成随机测试数据并保存二进制
import numpy as np
import os

# 确保输入目录存在
os.makedirs('input', exist_ok=True)

# 随机生成输入数据
x = np.random.randint(0, 100, size=(8, 2048)).astype(np.float16)
y = np.random.randint(0, 100, size=(8, 2048)).astype(np.float16)

# 保存输入数据
np.save('input/input_x.npy', x)
x.tofile('input/input_x.bin')

np.save('input/input_y.npy', y)
y.tofile('input/input_y.bin')

# 计算乘法真值并保存
z = x * y
z.tofile('input/golden.bin')
